package com.heima.wemedia.service.Impl;

import com.alibaba.fastjson.JSON;
import com.heima.aliyun.scan.GreenScan;
import com.heima.aliyun.scan.ScanResult;
import com.heima.common.constants.message.PublishArticleConstants;
import com.heima.common.constants.wemedia.WemediaConstants;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.service.WmNewsAutoScanService;
import com.heima.wemedia.service.WmNewsService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.AmqpException;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author mrchen
 * @date 2022/7/7 14:47
 */
@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {
    @Autowired
    private WmNewsService wmNewsService;
    @Value("${file.oss.web-site}")
    private String webSite;
    @Autowired
    private AdminFeign adminFeign;
    @Autowired
    private GreenScan greenScan;
    @Autowired
    private RabbitTemplate rabbitTemplate;



    @Override
    public void autoScanWmNews(Integer id) {
        // 1. 参数校验 (id不为空  )
        if(id == null){
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID,"id为空，审核失败");
        }
        // 2. 根据id查询文章数据
        WmNews wmNews = wmNewsService.getById(id);
        if (wmNews == null) {
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"对应文章信息不存在");
        }
        // 3. 判断文章状态是否为 1 (待审核)
        if (!WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(wmNews.getStatus())){
            CustException.cust(AppHttpCodeEnum.DATA_NOT_ALLOW,"文章状态不为待审核，无需审核");
        }
        // 4. 提取文章中所有的文本内容 和 图片内容路径
        Map<String,Object> textAndImages = handleTextAndImages(wmNews);
        String content = (String)textAndImages.get("content");
        List<String> images = (List<String>)textAndImages.get("images");
        // 5. 进行自管理敏感词校验
        boolean handleSensitive = handleSensitive(content,wmNews);
        if(!handleSensitive){
            log.info(" 文章中 包含敏感词，敏感词审核未通过 ==================");
            return;
        }
        // 6. 进行阿里云内容安全文本检测
        boolean textScan = handleTextScan(content,wmNews);
        if(!textScan){
            log.info(" 文章中 包含违规词汇，阿里云文本审核未通过 ==================");
            return;
        }
        // 7. 进行阿里云内容安全图片检测
        if(!CollectionUtils.isEmpty(images)){
            boolean imagesScan = handleImagesScan(images,wmNews);
            if(!imagesScan){
                log.info(" 文章中 包含违规图片，阿里云图片审核未通过 ==================");
                return;
            }
        }
        // 8. 如果全部检测通过  将状态修改为8 (通过)
        updateWmNews(wmNews,WmNews.Status.SUCCESS.getCode(), "审核通过");

        // 9. TODO 发送延时消息，进行文章发布上架操作
            //1. 获取发布时间
            long publishTime = wmNews.getPublishTime().getTime();
            //2. 获取当前时间
            long nowTime = System.currentTimeMillis();
            //3.距离发布时间=发布时间-当前时间
            long remainTime = publishTime - nowTime;
            //4.发送消息，并设置消息头 x-delay ：距离发布时间
            rabbitTemplate.convertAndSend(
                    PublishArticleConstants.DELAY_DIRECT_EXCHANGE,
                    PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY,
                    wmNews.getId(),
                    new MessagePostProcessor() {
                        /**
                         *      消息后置处理器  用于改变消息属性
                         * @param message
                         * @return
                         * @throws AmqpException
                         */
                        @Override
                        public Message postProcessMessage(Message message) throws AmqpException {
                            message.getMessageProperties()
                                    .setHeader("x-delay",remainTime<=0?0:remainTime);  //如果这个时间小于等于零就取零 相反就取这个时间
                            return message;
                        }
                    }
            );
            log.info("成功发送文章代发布消息，  文章id：{}   延时时间：{} ",wmNews.getId(),remainTime);
    }



    /**
     * 调用阿里云图片审核
     * @param images  待审核图片内容
     * @param wmNews   文章对象
     * @return
     */
    private boolean handleImagesScan(List<String> images, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.imageUrlScan(images);
            // pass（通过下一步）  review（3 人工审核）  block (2 失败)
            String suggestion = scanResult.getSuggestion();
            switch (suggestion){
                case "block":
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(), "图片中包含违规信息，审核未通过: "+scanResult.getLabel());
                    break;
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),  "图片中包含不确定因素，转为人工审核: "+scanResult.getLabel());
                    break;
                case "pass":
                    flag = true;
                    break;
                default:
                    // 转为人工审核
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(), "阿里云审核出现异常，转为人工审核");
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 转为人工审核
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(), "阿里云审核出现异常，转为人工审核");
        }
        return flag;
    }

    /**
     * 调用阿里云文本审核
     * @param content  待审核文本内容
     * @param wmNews   文章对象
     * @return
     */
    private boolean handleTextScan(String content, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.greenTextScan(content);
            // pass（通过下一步）  review（3 人工审核）  block (2 失败)
            String suggestion = scanResult.getSuggestion();
            switch (suggestion){
                case "block":
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(), "文本中包含违规词汇，审核未通过: "+scanResult.getLabel());
                    break;
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),  "文本中包含不确定因素，转为人工审核: "+scanResult.getLabel());
                    break;
                case "pass":
                    flag = true;
                    break;
                default:
                    // 转为人工审核
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(), "阿里云审核出现异常，转为人工审核");
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 转为人工审核
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(), "阿里云审核出现异常，转为人工审核");
        }
        return flag;
    }

    /**
     * 敏感词审核
     * @param content  文本内容
     * @param wmNews    文章对象
     * @return
     */
    private boolean handleSensitive(String content, WmNews wmNews) {
        // 1. 基于feign远程查询敏感词列表
        ResponseResult<List<String>> sensitives = adminFeign.sensitives();
        if (!sensitives.checkCode()) {
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR);
        }
        List<String> sensitivesData = sensitives.getData();
        // 2. 基于DFA算法，封装数据模型
        SensitiveWordUtil.initMap(sensitivesData);
        // 3. 基于DFA算法，检测内容是否包含敏感词
        Map<String, Integer> map = SensitiveWordUtil.matchWords(content);
        // 4. 如果包含敏感词 将文章状态修改为 2 (审核失败)
        if(!CollectionUtils.isEmpty(map)){
            updateWmNews(wmNews,WmNews.Status.FAIL.getCode(), "内容中包含敏感词: " + map);
            return false;
        }
        return true;
    }
    private void updateWmNews(WmNews wmNews,Short status,String reason){
        wmNews.setStatus(status);
        wmNews.setReason(reason); // {冰毒: 1}
        wmNewsService.updateById(wmNews);
    }

    /**
     * 抽取文章中 所有的文本内容 和 图片内容
     * @param wmNews
     * @return
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {
        List<Map> contentMap = JSON.parseArray(wmNews.getContent(),Map.class);
        // 1. 将所有文本内容拼接成一个字符串
        //  我的家乡很美_hmtt_国家很强大
        String content = contentMap.stream()
                // 找到type为text对应的数据
                .filter(m -> "text".equals(m.get("type")))
                // 获取对应value的值
                .map(m -> String.valueOf(m.get("value")))
                // 将字符串使用 _hmtt_进行拼接
                .collect(Collectors.joining("_hmtt_"));
        // 和标题拼接在一起
        content = content + "_hmtt_" + wmNews.getTitle();
        // 2. 将所有图片路径收集到一个集合中

        // 收集内容中引用的图片
        List<String> images = contentMap.stream()
                .filter(m -> "image".equals(m.get("type")))
                .map(m -> String.valueOf(m.get("value")))
                .collect(Collectors.toList());
        // 判断封面字段是否为空
        if(StringUtils.isNotBlank(wmNews.getImages())){
            //   url1,url2,url3
            List<String> coverImages = Arrays.stream(wmNews.getImages().split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());
            images.addAll(coverImages);
        }
        // 去除重复的图片
        images = images.stream().distinct().collect(Collectors.toList());
        // 3. 将内容和路径集合封装到map返回
        Map result = new HashMap();
        result.put("content",content);
        result.put("images",images);
        return result;
    }
}
